//*************************************************************************************************
//Copyright (c) 2015.year. @hjjer.
//If you are a hjjer(hejianjiao.org), This code is unlicensed, otherwise,is licensed under the GPL.
//*************************************************************************************************

package com.truthbean.ssm.core.web.kindeditor;

import com.alibaba.fastjson.JSONObject;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * @author Truthbean
 * @since 2015-2-18 16:14:30
 */
@WebServlet("/file-manager")
public final class FileManager extends HttpServlet {

    private static final long serialVersionUID = 1L;

    @Override
    public void service(final HttpServletRequest request, final HttpServletResponse response)
            throws IOException, ServletException {

        PrintWriter out = null;

        try {
            response.setContentType("text/html; charset=UTF-8");
            out = response.getWriter();

            out.write("\r\n");
            out.write("\r\n");
            out.write("\r\n");
            out.write("\r\n");

            // 根目录路径，可以指定绝对路径，比如 /var/www/resource/
            String rootPath = request.getServletContext().getRealPath("/")
                    + "resource/";
            // 根目录URL，可以指定绝对路径，比如 http://www.yoursite.com/resource/
            String rootUrl = request.getContextPath() + "/resource/";
            // 图片扩展名
            String[] fileTypes = new String[]{"gif", "jpg", "jpeg", "png", "bmp"};

            String dirName = request.getParameter("dir");
            if (dirName != null) {
                if (!Arrays.asList(
                        new String[]{"image", "flash", "media", "file"})
                        .contains(dirName)) {
                    out.println("Invalid Directory name.");
                    return;
                }
                rootPath += dirName + "/";
                rootUrl += dirName + "/";
                File saveDirFile = new File(rootPath);
                if (!saveDirFile.exists()) {
                    saveDirFile.mkdirs();
                }
            }
            // 根据path参数，设置各路径和URL
            String path = request.getParameter("path") != null ? request
                    .getParameter("path") : "";
            String currentPath = rootPath + path;
            String currentUrl = rootUrl + path;
            String currentDirPath = path;
            String moveupDirPath = "";
            if (!"".equals(path)) {
                String str = currentDirPath.substring(0,
                        currentDirPath.length() - 1);
                moveupDirPath = str.lastIndexOf("/") >= 0 ? str.substring(0,
                        str.lastIndexOf("/") + 1) : "";
            }

            // 排序形式，name or size or type
            String order = request.getParameter("order") != null ? request
                    .getParameter("order").toLowerCase() : "name";

            // 不允许使用..移动到上一级目录
            if (path.contains("..")) {
                out.println("Access is not allowed.");
                return;
            }
            // 最后一个字符不是/
            if (!"".equals(path) && !path.endsWith("/")) {
                out.println("Parameter is not valid.");
                return;
            }
            // 目录不存在或不是目录
            File currentPathFile = new File(currentPath);
            if (!currentPathFile.isDirectory()) {
                out.println("Directory does not exist.");
                return;
            }

            // 遍历目录取的文件信息
            List<HashMap> fileList = new ArrayList<>();
            if (currentPathFile.listFiles() != null) {
                for (File file : currentPathFile.listFiles()) {
                    HashMap<String, Object> hash = new HashMap<>();
                    String fileName = file.getName();
                    if (file.isDirectory()) {
                        hash.put("is_dir", true);
                        hash.put("has_file", (file.listFiles() != null));
                        hash.put("filesize", 0L);
                        hash.put("is_photo", false);
                        hash.put("filetype", "");
                    } else if (file.isFile()) {
                        String fileExt = fileName.substring(
                                fileName.lastIndexOf(".") + 1).toLowerCase();
                        hash.put("is_dir", false);
                        hash.put("has_file", false);
                        hash.put("filesize", file.length());
                        hash.put("is_photo", Arrays.asList(fileTypes)
                                .contains(fileExt));
                        hash.put("filetype", fileExt);
                    }
                    hash.put("filename", fileName);
                    hash.put("datetime", new SimpleDateFormat(
                            "yyyy-MM-dd HH:mm:ss").format(file.lastModified()));
                    fileList.add(hash);
                }
            }

            if (null != order) {
                switch (order) {
                    case "size":
                        Collections.sort(fileList, new SizeComparator());
                        break;
                    case "type":
                        Collections.sort(fileList, new TypeComparator());
                        break;
                    default:
                        Collections.sort(fileList, new NameComparator());
                        break;
                }
            }
            JSONObject result = new JSONObject();
            result.put("moveup_dir_path", moveupDirPath);
            result.put("current_dir_path", currentDirPath);
            result.put("current_url", currentUrl);
            result.put("total_count", fileList.size());
            result.put("file_list", fileList);

            response.setContentType("application/json; charset=UTF-8");
            out.println(result.toJSONString());

            out.write('\r');
            out.write('\n');
        } catch (Throwable t) {
            if (out != null) {
                if (response.isCommitted()) {
                    out.flush();
                } else {
                    out.close();
                }
            }

        } finally {
            if (out != null) {
                out.close();
            }
        }
    }

    public class NameComparator implements Comparator {

        @Override
        public int compare(Object a, Object b) {
            HashMap hashA = (HashMap) a;
            HashMap hashB = (HashMap) b;
            if (((Boolean) hashA.get("is_dir")) && !((Boolean) hashB.get("is_dir"))) {
                return -1;
            } else if (!((Boolean) hashA.get("is_dir")) && ((Boolean) hashB.get("is_dir"))) {
                return 1;
            } else {
                return ((String) hashA.get("filename")).compareTo((String) hashB.get("filename"));
            }
        }
    }

    public class SizeComparator implements Comparator {

        @Override
        public int compare(Object a, Object b) {
            HashMap hashA = (HashMap) a;
            HashMap hashB = (HashMap) b;
            if (((Boolean) hashA.get("is_dir")) && !((Boolean) hashB.get("is_dir"))) {
                return -1;
            } else if (!((Boolean) hashA.get("is_dir")) && ((Boolean) hashB.get("is_dir"))) {
                return 1;
            } else if (((Long) hashA.get("filesize")) > ((Long) hashB.get("filesize"))) {
                return 1;
            } else if (((Long) hashA.get("filesize")) < ((Long) hashB.get("filesize"))) {
                return -1;
            } else {
                return 0;
            }
        }
    }

    public class TypeComparator implements Comparator {

        @Override
        public int compare(Object a, Object b) {
            HashMap hashA = (HashMap) a;
            HashMap hashB = (HashMap) b;
            if (((Boolean) hashA.get("is_dir")) && !((Boolean) hashB.get("is_dir"))) {
                return -1;
            } else if (!((Boolean) hashA.get("is_dir")) && ((Boolean) hashB.get("is_dir"))) {
                return 1;
            } else {
                return ((String) hashA.get("filetype")).compareTo((String) hashB.get("filetype"));
            }
        }
    }
}
